add solution and test to diet challenge