#4 add a few functions