[WIP] Add test code

#10