Some minor updates

minor fixes to make it compile