added edit
fixed bug

#1 resolved
#2
Merge pull request #4 from will-harrison/master

RFE complete