Added feature to change level of difficulty