Adding save behaivor