Change button class based on shell.