made code a bit prettier