Minor Changes.

Removed the tests from the constructor, started to add in the
difficulty, and fixed changing back the taken square.