Code update

Updated code made during the competition.