tidy up wrapper code