working on code...minor change