Created solution to [easy]challange#1 using Python