worked on editing feature a little