Refactor code to transformers convention

#3