Added rendering code