yeah i didn't want to change the other code when I changed this class