Improve menu #WIP