Adding basic driving code