added feature as arg to main function + added basic parameter sweep