Add some initial tests (Ginkgo specs = BDD-style)

Still doesn't support diagonals.
Simplify main.go by moving things into functions

Still doesn't support diagonals.