Small fixes see code