Finished with assignment 2
Added comments. Adding 'enter' key handler later