removed print, added some tests (to finish)