add battlements and tidy code a little