My friend helped me fix some code