Added entity as a dependancy