Fixes #3 styling changes