fix program and add new solution