Update
Added test class