correct initial code

By mistake, first uploaded codes were actually older version. This
version is the right up-to-date version.