Bug fix
Updated flat button style