Both buttons now function