Updated to take an command line parameter to choose which solution to run