Lambda Practice Problems Python