Finished the first couple of tasks. 
Introduced bug where it's not possible to return to the main menu after
first task have been run.
Finished the assignment