Added a nice menu stage; code is a mess though