End of Section 1 challenge

This is my contribution to the end of section 1 challenge :slight_smile: