Updated code to finalize project