Slight change to begin code