added some more code to check orientation

and drive wheels accordingly