Updated driver to use correct pins