final project code improvements