Remove items from list based on condition python