7 add testing

For issue #7 "Add testing"