Added puzzle method and more styling