updating the code to match my environment