Add integer argument handling