small fixes

fixes to make it look okay for demo tomorrow