Problem2: adjusted python style