I just added in two buttons