Minor code cleanup
Fix broken layout on resize: use flex-box
Improved design