Difficulty prompt and comments

Added changes by SB to support difficulty selection and also some code
comments