Improve design and make it responsive