Fixed some bugs and complete milestone add, update and remove