added new feature shit