Program solves multiple test cases, needs more tests