Added new update method &&
Cleaned up test and added new update test