Minor change, now less bugs
I think this works out all the kinks