Added drive class. Still haven't compiled. Will do so and debug later