Updated after meeting with Prof yang