Added the use statement for the entities