Start to add some basic tests