made changes to start and end screen