the 2 looks a little magic. maybe consider passing it as a kwarg? and/or making it a constant