small changes

made small changes to both files