\documentclass[11pt]{article}
\usepackage{fullpage}
\usepackage{amsmath,amsfonts,amsthm,amssymb}
\usepackage{url}
\usepackage[demo]{graphicx}
\usepackage{caption} 
\usepackage{algpseudocode}
\usepackage{bbm}
\usepackage{float}
\usepackage{framed}
\usepackage{enumerate}
\usepackage{color}
\usepackage[colorlinks=true, linkcolor=red, urlcolor=blue, citecolor=blue]{hyperref}

\DeclareMathOperator*{\E}{\mathbb{E}}
\let\Pr\relax
\DeclareMathOperator*{\Pr}{\mathbb{P}}
\DeclareMathOperator*{\R}{\mathbb{E}}

\topmargin 0pt
\advance \topmargin by -\headheight
\advance \topmargin by -\headsep
\textheight 8.9in
\oddsidemargin 0pt
\evensidemargin \oddsidemargin
\marginparwidth 0.5in
\textwidth 6.5in

\parindent 0in
\parskip 1.5ex

\newcommand{\homework}[2]{
	\noindent
	\begin{center}
		\framebox{
			\vbox{
				\hbox to 6.50in { {\bf COMPSCI 514: Algorithms for Data Science} \hfill Fall 2019 }
				\vspace{4mm}
				\hbox to 6.50in { {\Large \hfill Problem Set #1  \hfill} }
				\vspace{2mm}
				\hbox to 6.50in { {Group: #2 \hfill} }
			}
		}
	\end{center}
	\vspace*{4mm}
}

\begin{document}
	
	\homework{1}{Alan Turing, Ada Lovelace, Carl Gauss}
	
	One member of each group should submit the problem set. Make sure to mark all group members on the assignment in \href{https://www.gradescope.com/}{GradeScope}.
	
	\section*{Problem 1}
	
	In Latex you can easily include written explanations, equations, and images as part of your solutions. For example:
	
	\begin{align*}
	\Pr \left [ X \ge 2 \E[X] \right ] \le \frac{1}{2}.
	\end{align*}
	
	Or an image:
	
	\begin{figure}[h]
		\centering
		\includegraphics[width=0.4\textwidth]{image.jpg}
		\caption{This is an example image. Describe any images with sufficient detail in the caption.}
	\end{figure}
	
	You may also want to write pseudocode to describe an algorithm:
	
	\medskip
	
	\begin{algorithmic}
		\State $val \gets 1$
		\State $sum \gets 0$
		\For {$i = 1,\ldots,100$}
		\If {$(i \mod 10) == 0$}
		\State $sum \gets sum + val$
		\EndIf
		\State {$val \gets val + i$}
		\EndFor
		\State \Return $sum$
	\end{algorithmic}
	
	
	\section*{Problem 2}
	
	For multipart problems you can use subsections.
	
	\subsection*{Problem 2-a}
	
	First part here.
	
		\subsection*{Problem 2-b}
	
	Second part here.
	
	
\end{document}