Control number of posts on home page and category

Home & Category Number

function hwl_home_pagesize( $query ) {
    if ( is_admin() || ! $query->is_main_query() )
        return;

    if ( is_home() ) {
        // Display only 3 posts 
        $query->set( 'posts_per_page', 3 );
        return;
    }    

if ( is_archive() ) {
        // Display only 5 posts 
        $query->set( 'posts_per_page', 5 );
        return;
    }
}
add_action( 'pre_get_posts', 'hwl_home_pagesize', 1 );