Implement working with EditPart (fixes #629)