Updated Code Styling