A little cleanup :)
Menu complete
forgot to save