edit milestone functionallity working - fix