Removed TODO comment (which has already been addressed)