Fixed issue with edit and delete. Added shitty UI for testing.