Here's my code, it's a bit messy rn but you can check if I am doing anything wrong