Fixes #7 - Use relative import