Button functionality added, -Pepe