Updated code for minor edits.