[Update]: cleaned code.