Finished assignment

fixing it up making it look pretty