New update

Fixed code