Modify screenToWorld to handle argument count

Author: perminder-17

src/core/environment.js

@@ -1316,14 +1316,14 @@ function environment(p5, fn, lifecycles){
   fn.screenToWorld = function(screenPosition) {
     // remember how many components the caller actually provided to createVector
     const origDimension = this.dimension;
-  
+    const screenToWorldArgs = arguments.length;
     if (typeof screenPosition === 'number') {
       // We got passed numbers, convert to vector
       screenPosition = this.createVector(...arguments);
     }
-  
+
     const matrix = this._renderer.getWorldToScreenMatrix();
-    if (origDimension === 2) {
+    if (origDimension === 2 || screenToWorldArgs === 2) {
       // Calculate a sensible Z value for the current camera projection that
       // will result in 0 once converted to world coordinates
       let z = matrix.mat4[14] / matrix.mat4[15];