TODO small changes