I just can't seem to get this class right :/