I need help improving my code. It's a basic game where an object avoids stuff coming at it$.$I need to improve the graphics using the same method I have been using to draw, I can't load an image. A submarine is the main object that will be avoiding stuff coming at$.$The objects coming at it should be a tire, a fish$/$shark$,$ anything thing else you think will look good. Try your best to create the best graphics you can. The game logic is all good. Take your time drawing it$,$ it should look realistic, change the background to blue for water. I want the tire to actually look like a tire and not just a red blob. A submarine is hard to draw but use a draw a yellow sideways oval and a gray pipe sticking out the top. Google a submarine to imagine what it looks like. Draw a fish so you actually know it$$s a fish not just a gray block. I$$m really emphasizing how important it is to take your time, this will take time and if you try to rush it$,$ it will come out bad. As realistic as possible using pygame.draw and using different shapes if you can.

import sys

import random

# Initialize Pygame

pygame.init$()$

# Constants

WIDTH, HEIGHT $=800,600$

SUBMARINE$\_$WIDTH, SUBMARINE$\_$HEIGHT $=50,30$

OBSTACLE$\_$WIDTH, OBSTACLE$\_$HEIGHT $=50,50$

FPS $=60$

SUBMARINE$\_$SPEED $=5$

# Colors

WHITE $=(255,255,255)$

BLUE $=(0,0,255)$

RED $=(255,0,0)$

# Game variables

submarine$\_$rect $=$pygame$.$Rect$($WIDTH $//2-$SUBMARINE$\_$WIDTH $//2,$HEIGHT $//2-$SUBMARINE$\_$HEIGHT $//2,$SUBMARINE$\_$WIDTH, SUBMARINE$\_$HEIGHT$)$

obstacle$\_$rect $=$pygame$.$Rect$($WIDTH$,$random$.$randint$(0,$HEIGHT $-$OBSTACLE$\_$HEIGHT$),$OBSTACLE$\_$WIDTH, OBSTACLE$\_$HEIGHT$)$

score $=0$

# Create the screen

screen $=$pygame$.$display.set$\_$mode$(($WIDTH$,$HEIGHT$))$

pygame.display.set$\_$caption$("$Submarine Game"$)$

# Fonts

font $=$pygame$.$font.Font$($None$,36)$

# Functions

def draw$\_$submarine$()$:

$$ pygame.draw.rect$($screen$,$BLUE$,$ submarine$\_$rect$)$

def draw$\_$obstacle$()$:

$$ pygame.draw.rect$($screen$,$RED$,$ obstacle$\_$rect$)$

def draw$\_$score$()$:

$$ score$\_$text $=$font$.$render$("$Score: $"+$str$($score$),$True$,$ WHITE$)$

$$ screen.blit$($score$\_$text, $(10,10))$

# Game loop

clock $=$pygame$.$time.Clock$()$

while True:

$$ for event in pygame.event.get$()$:

$$ if event.type $==$pygame$.$QUIT:

$$ pygame.quit$()$

$$ sys$.$exit$()$

$$ keys $=$pygame$.$key.get$\_$pressed$()$

$$ if keys$[$pygame$.$K$\_$UP$]$and submarine$\_$rect.top $>0$:

$$ submarine$\_$rect.y $-=$SUBMARINE$\_$SPEED

$$ if keys$[$pygame$.$K$\_$DOWN$]$and submarine$\_$rect.bottom $<$HEIGHT:

$$ submarine$\_$rect.y $+=$SUBMARINE$\_$SPEED

$$ # Update obstacle position

$$ obstacle$\_$rect.x $-=5$

$$ if obstacle$\_$rect.right $<0$:

$$ obstacle$\_$rect.x $=$WIDTH

$$ obstacle$\_$rect.y $=$random$.$randint$(0,$HEIGHT $-$OBSTACLE$\_$HEIGHT$)$

$$ score $+=1$

$$ # Check for collisions

$$ if submarine$\_$rect.colliderect$($obstacle$\_$rect$)$:

$$ print$("$Game Over!"$)$

$$ pygame.quit$()$

$$ sys$.$exit$()$

$$ # Draw everything

$$ screen.fill$($WHITE$)$

$$ draw$\_$submarine$()$

$$ draw$\_$obstacle$()$

$$ draw$\_$score$()$

$$ # Update display

$$ pygame.display.flip$()$

$$ # Cap the frame rate

$$ clock.tick$($FPS$)$