Python Basic Certificate Hackerrank Solution