Python Get Additional Info Hackerrank Solution