I' m stupid i dont know what i'm doing

This is a simple update to your fancy code!