Adding further gradient checks