updated class/TODO stuff