fixed X and X_test not being normal lists (arrays)