Improve styling and add responsive layout