import math


def work(force, displacement, theta):
    """
       Calculates the work done by 
       the force while displacing
       an object at an angle theta
       
       Parameters
       ----------
       force : float
       displcement : float 
       theta : float
       
       Returns
       -------
       float
    """
    return force * displacement * math.cos(theta)