Python Solution hacker rank