Added driver test code and solution