Now everything I've done so far is part of a function: problem_3b