added dependency and changed class