[ UPDATE ] - Small layout change