Created driver class in seperate file