added difficulty choice screen