Solution for 2_homework_cheater