updating code to submit for solutions-engineering interview