Added first unittest
Fixed formatting