Implemented difficulty based behavior

Still need to add it as a config option