python if else hackerrank solution github