Forgot some entities and added dependency to the gradle fiels