Slight changes to Training and session code