Fix some styles
Update the gallery directive