Update main js
Experiment with flex layout