better menu
code improvement
added replay function