Python Function Input Variable Of A Cosine