would be good to call the correct class