Added draw function