Cleanup edit privilege functionality

Signed-off-by: Madhura Jayaratne <madhura.cj@gmail.com>