поправил и добавил трейты