Started work on adding difficulty buttons