Added another unit test + formatted code