Fixes #3 Missing module imports