Minor refactor.

Im a moron, this is not the Oscar's...
First touches of AI