Actually works

Removed unnecessary imports and now it actually works