Added toy example code, code cleaned