started writing proper tests