overwritten trajectory loader now passes kwargs properly