editing my code