Add driver registers and cleanup