Add changes to call class