Added difficulty screen
Made it more object-oriented, takes fewer lines and less duplicates