changed my code