Feature/clean up

@shanejonas @stryderjzw @sophlee r?