changing button type
Updating tests