Button functionality and finishing touches