update setup to work with package

Signed-off-by: Steven Armstrong <steven.armstrong@inf.ethz.ch>