added more tests, fixed bug with relative import