add axis and better formatting