Some minor changes, I don't know how to do functions properly
styling