Logic works, needs styling