Adding tests and making sure they pass

Still not sure if this is the right direction to be going, but
it seems somewhat reasonable