enhance my code