(WIP) Add some automated tests

This is no where enough tests yet but some is better than none!