matrix script hackerrank solution