Please update the code comment too? :)