updated to use correct equation
updated to use correct equation