Add arguments to support updating source paths

Update comments

Fix indentation