Updating draw code