UI Change
Remove masonry
Remove __init__