Change features flow, use the same model with post

Change features flow, use the same model with post