update final code
