Trying to fix code after list problem.