Added some basic tests
Gotta clean it all up when everything works perfect