Fixed bugs in solver
Added driver script for problem 1