Fix #2 - Add a customizable unit