Fixed a little of my code