Added drawLine code