Formatting and added another test condition.