Changed code + changed readme