New feature: User can set Code