- Highlight features in readme - Add simple examples per ML type in readme - Improve example code - Consider using sphinx