Cleanup and formatting

Signed-off-by: DziNeIT <dzineit@live.co.uk>