Fixing up MenuState - started on levels logic