Refactored and cleaned the code, removed setPlayerController
Cleaned up imports