Added restart button and key handling. Closes #49.