Finalizing code and styling