Final Project - Fixed Undo and main screen

need to javadoc