Updated code: improve code annotation
Updated code annotation