Add some tests. Finally!