Added difficulty, and win & lose situation check