Added update method
Updated model tests