put finishing touches on my code