Added import, edited entity class name