Updated readme accordingly to updated code