Solution for week7
Week08 WIP