Cleanup
Fix parameter checking