Added some documentation and tidies up. Now foodcritic compliant