Added difficulty option

Warning: Code has not yet been tested, the program will most likely crash/work incorrectly