Updated to match README