lone_sum logic problem as a list comprehension