Added more stuff to the menu and to die function