Single Line Function In Python Practice