little bit of polish for the last demo