For One Liner Python