Added some rendering code