add some missing pieces

Signed-off-by: Cristian Stoica <crististm@gmail.com>