Transition when drawing green square
Cleaned code