Added class to create/update button