Fix import paths. Closes #2