Made layout work the way I wanted it to