Some commenting
Added some todos, added a general callback for screenstate change