[Front-end] Removed flex box, made it more responsive