Python Apply Function With Multiple Arguments