Updated code this time.