cleanup code to look better on page