first edit of my code