One line python solution