assignment part 1 complete. all driver code working