Modified something
Added test