saving state - I dont hate how this is coming together :)