Added setup and draw methods to driver