Added missing update and draw functions