Corrected formatting, added comment on unit test