Fix up getDirection, and imports in 2 TE classes

To make it compile with newer Forge versions
Merge pull request #4 from Alex-hawks/master

Fix up getDirection and imports in 2 TE Classes