Add tests to distribution

See #119