Add more parameter checking