Added retrieving of test and train messages
Added test bed