Updated annotations to code