Better handling input parameter