Closes #5 with more styling