Minor
Adding test