Clearing up TODO