updated some code, something else I'm sure