Add a screen class. Fix the parameters which broke. Add a restart button shift+R.