Cleaned up repo and added some tests

Yo.