Added main menu, and trying to beautify things