Added tests! Very first version of tests!