some updates in display and draw functions