Added setSource ability, and altered tests to make more sense and not based around 10x10 grid sizes