minor edits and adding in a variable