Implemented AI class layer to control active behaviour