some fixes and bugs))