Fixing easy mode as well