Python code - 29 lines - codepad