Modifed to have test scheme in commandline