Was pretty easy cause I copied code from an earlier problem.