Update controller.js

Add support for shadow in freeDrawing mode
Missing `return`?