Homework is working, added another layout for button placement