Added 'known y' training case
Simplified test function, added unknown-y case to test and predict functions