Refactored stages and added menu/lives etc.