Formatting fix and add test