Refactored testing, now uses stretchr/testify/assert package. Also changed changed grid to a type rather than a struct.