Adding to toString to class