added save function thing...

Still not working right though