Add basic tests
Update build