Reformatted code
Added comment to function
Improved unit test by using small part of real data