correct code
done with this shit