Adding styling back

Post logic debugging, styling reapplied.
First unit test working

Basic unit test working.