some formatting
Fixed #17 and made theta max a constant.
cleaned up.