update code

Dec 24th, happy Christmas Eve