Code Challenge 1 for CS225