Switched to using tile entity and imported the models