Makes the edit button work correctly
Merge pull request #122 from TheTeam8s/feature-editButton

Makes the edit button work correctly