Converted rendering to accept entity in update/draw methods