adding logic class