Tidy up Entity