Could you please write some tests?