Some button functionality added