Added test to clarify expectations, and styled code per Ruby Style Guides