minor main script changes