Add support for more generated file types