Finalizing functionality, added some styling