Update entity.cpp