Trying to fix some stuff with the menu