testing code for the 1st time
modification after testing -1
with new predict function