Add: started to write tests