Added edit info feature