Added restart method. Need to clean up code and work on beautifying front end.