Changed small code