Added comments
and fixed keyboard/mouse control