Added MLlib tests
Small bug fixes