Add if statement for limit of 3