Improve add entity