* Changed the indentation a bit and added a header.
* Class name is added as in parameter to masonry function.