cleaned up code
added a start button