Hackerrank Python Write A Function Solution