Fix styling
Add more screenshots