Add masonry, tidier js