Added and tested MLlib tests