Added main menu (pretty ugly but it's in)