Add more changes on Update