Add tests

I apologize this took so long on this module.