добавил entity, небольшие фиксы