Add tests

Waiting travis tests ...