\documentclass[11pt, oneside]{article}   	% use "amsart" instead of "article" for AMSLaTeX format
\usepackage{geometry}
\usepackage[]{algorithm2e}
\geometry{letterpaper}
\usepackage{graphicx}
\usepackage{amssymb}
\usepackage{amsthm}
\usepackage{parskip}

\graphicspath{ {images/} }

\newtheorem*{theorem}{Theorem}
\newtheorem*{lemma}{Lemma}
%SetFonts

\title{CS340 - Problem Set 1}
\author{YOUR NAME}
%\date{}							% Activate to display a given date or no date

\begin{document}
\maketitle

\section{Problem A}
\subsection{Description}
Description goes here.

\subsection{Pseudocode}
\begin{algorithm}
\SetKwFunction{foo}{foo}
\SetKwProg{Fn}{Function}{}{}
\Fn{\foo{arg1, list1, list2}}{
  \If{arg1} {
    \Return arg1
  } \Else {
    \For{e $\in$ list1} {
      do this
    }
    flag = True

    \While{flag} {
      flag = False
    }
  }
}
\end{algorithm}

\subsection{Time Analysis}
Time Analysis goes here. $O(n^2)$


\subsection{Proof of Correctness}
\begin{lemma}
Lemma goes here
\end{lemma}

\begin{proof}
Base Case:

Inductive Hypothesis:

Inductive Step
\end{proof}

\section*{Collaboration}
Name1, Name2, ...

\end{document}


