Modifications to improve age synthesis model