Added area function and tests