Changed answer to second exmaple