Driver works as it should now