check file types from the input directory

Signed-off-by: Ethan Vizitei <ethan.vizitei@gmail.com>