So reading the assignment for problem A, is this coding what my professor is looking for?